import selenium
from selenium import webdriver
#无头浏览器
from selenium.webdriver.chrome.options import Options
#规避检查
from selenium.webdriver import ChromeOptions
import xlrd
import time
from selenium.webdriver.common.keys import Keys
#打开浏览器方法
def open_browser():
    # 创建一个参数对象，用来控制chrome以无界面模式打开
    chrome_options = Options()
    # 无头浏览器
    chrome_options.add_argument('--headless')
    chrome_options.add_argument('--disable-gpu')
    # 规避检测浏览器
    option = ChromeOptions()
    option.add_experimental_option('excludeSwitches', ['enable-automation'])

    path = "./chromedriver.exe"
    browser = webdriver.Chrome(executable_path=path, options=option)
    return browser
# 获取数据方法
def get_data(browser,url):
    browser.get(url)
    data=[]
    #搜索结果截屏
    div_list=browser.find_elements_by_xpath('//*[@id="content_left"]/div[@tpl="se_com_default"]')
    i=0
    for div in div_list:
        title=div.find_element_by_xpath('./h3/a').text  #//*[@id="5"]/h3/a
        desc=div.find_element_by_xpath('./div[1]').text
        dic={
            'title':title,
            'sort':i+1,
            'desc':desc
        }
        i = i + 1
        data.append(dic)
    # browser.close()
    return data

#关闭浏览器方法
def browser_close(browser):
    browser.close()
# 导出报表方法

#读取关键字方法
def read_excel(file):
    wb=xlrd.open_workbook(filename=file)
    table=wb.sheets()[0]
    nrows=table.nrows
    word=[]
    for i in range(nrows):
        # print(table.row_values(i))
        keyword=table.cell_value(i,0)
        word.append(keyword)
    return word
# 对接接口方法

if __name__=="__main__":
    file="seo2.xlsx"
    words=read_excel(file)
    for word in words:
        #print(word)
        url="https://www.baidu.com/s?ie=utf-8&wd="+word
        browser=open_browser()
        result = get_data(browser,url)
        browser_close(browser)
        print(result)


